Ternary in Python